#!/usr/bin/python3
# -*- coding: utf-8 -*-

"""
# @version: v1.0
# @author : wlis
# @Email : 1259319710@qq.com
# @Project : g-carbon-bio
# @File : stocksh_service.py.py
# @Software: PyCharm
# @time: 2025/2/5 10:25
# @description : 
"""
from service.models.stock_sh import StockSh
from service.stock_base_service import StockBaseService


class StockShService(StockBaseService):
    def __init__(self):
        self.table_name_am = 'stocksh_am'
        self.table_name_pm = 'stocksh_pm'
        super().__init__()

    def get_distinct_dates_am(self):
        return super().get_distinct_dates(self.table_name_am)

    def get_am(self, _stock, current_time=None, current_ap=None):
        """获取上海当日成交量表中的股票信息

        参数:
        _stock_code: 股票代码
        current_time: 查询日期，默认为None
        current_ap: 上午或下午标识，默认为None

        返回:
        股票对象列表
        """
        return super().get(self.table_name_am, _stock, current_time, current_ap)

    def get_pm(self, _stock, current_time=None, current_ap=None):
        """获取上海盘后成交量表中的股票信息

        参数:
        _stock_code: 股票代码
        current_time: 查询日期，默认为None
        current_ap: 上午或下午标识，默认为None

        返回:
        股票对象列表
        """
        return super().get(self.table_name_pm, _stock, current_time, current_ap)



    def add_list_am(self, stock_list):
        """将股票数据添加到上海当日成交量表

        参数:
        stock_list: 股票对象列表
        """
        self.add_list(stock_list, self.table_name_am)

    def add_list_pm(self, stock_list):
        """将股票数据添加到上海盘后成交量表

        参数:
        stock_list: 股票对象列表
        """
        self.add_list(stock_list, self.table_name_pm)


    def update_list_am(self, stock_list):
        """更新上海当日成交量表中的股票数据

        参数:
        stock_list: 股票对象列表
        """
        self.update_list(stock_list, self.table_name_am)

    def update_list_pm(self, stock_list):
        """更新上海盘后成交量表中的股票数据

        参数:
        stock_list: 股票对象列表
        """
        self.update_list(stock_list, self.table_name_pm)
